package top.ddltech.algorithm.sort;

import top.ddltech.algorithm.util.StdRandom;
import top.ddltech.algorithm.util.Stopwatch;

/**
 * 排序比较
 */
public class SortCompare {

    public static double time(String alg, Comparable[] a) {
        Stopwatch timer = new Stopwatch();
        if (alg.equals("Insertion")) {
            InsertionSort.sort(a);
        }
        if (alg.equals("Selection")) {
            SelectionSort.sort(a);
        }
        if (alg.equals("Shell")) {
            ShellSort.sort(a);
        }
        if (alg.equals("Merge")) {
            MergeSort.sort(a);
        }
        if (alg.equals("Quick")) {
            QuickSort.sort(a);
        }
        if (alg.equals("Heap")) {
            HeapSort.sort(a);
        }
        return timer.elapsedTime();
    }

    public static double timeRandomInput(String alg, int N, int T) {
        // 使用算法alg将T个长度为N的数组排序
        double total = 0.0;
        Double[] a = new Double[N];
        for (int t = 0; t < T; t++) {
            // 进行一次测试（生成一个数组并排序）
            for (int i = 0; i < N; i++) {
                a[i] = StdRandom.uniform();
            }
            total += time(alg, a);
        }
        return total;
    }

    /**
     * 进行两个排序算法的比较
     *
     * @param alg1 第一个算法名称
     * @param alg2 第二个算法名称
     * @param N    数组长度
     * @param T    重复次数
     */
    public static void compare(String alg1, String alg2, int N, int T) {
        // 算法1的总时间
        double t1 = timeRandomInput(alg1, N, T);
        // 算法2的总时间
        double t2 = timeRandomInput(alg2, N, T);
        System.out.printf("For %d random Doubles\n    %s is", N, alg1);
        System.out.printf(" %.1f times faster than %s\n", t2 / t1, alg2);
    }
}
